#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Create by zhang
# Create on 2022/6/27 21:51
from domain.report.dto.balance_sheet_dto import BalanceSheetSchema
from domain.report.extractor.balance_sheet.consolidated_balancesheet_filter import ConsolidatedBalanceSheetFilter
from domain.report.common.extract_content_enum import ExtractContentEnum
from domain.report.extractor.request import TaskData


class BalanceSheetFilter(ConsolidatedBalanceSheetFilter):
    def __init__(self):
        super(BalanceSheetFilter, self).__init__()

    def init(self):
        self.extract_type = ExtractContentEnum.BalanceSheet
        self.exclude_type = [ExtractContentEnum.ConsolidatedBalanceSheet]
        self.result = TaskData(name=self.extract_type.name)
        self.__class__.item_titles = BalanceSheetSchema.get_all_titles()


